Skip to content

Implementing sampling stats#550

Merged
willow-ahrens merged 15 commits into
mainfrom
rs/implementing_sampling_stats
Jul 24, 2026
Merged

Implementing sampling stats#550
willow-ahrens merged 15 commits into
mainfrom
rs/implementing_sampling_stats

Conversation

@Ruchika-2003

Copy link
Copy Markdown
Collaborator

No description provided.

@codspeed-hq

codspeed-hq Bot commented Jul 3, 2026

Copy link
Copy Markdown

Merging this PR will not alter performance

✅ 24 untouched benchmarks


Comparing rs/implementing_sampling_stats (196381c) with main (ddf4ffc)

Open in CodSpeed

@willow-ahrens
willow-ahrens marked this pull request as draft July 3, 2026 21:35
@willow-ahrens

Copy link
Copy Markdown
Member

@Ruchika-2003 can we get this ready to go?

@Ruchika-2003

Copy link
Copy Markdown
Collaborator Author

@Ruchika-2003 can we get this ready to go?

I removed the heuristic flag as we discussed, the current one with those changes is ready to go

@willow-ahrens

Copy link
Copy Markdown
Member

@Ruchika-2003 fix precommit and resolve conflicts

@willow-ahrens
willow-ahrens marked this pull request as ready for review July 24, 2026 17:12
@willow-ahrens
willow-ahrens enabled auto-merge (squash) July 24, 2026 17:13

@willow-ahrens willow-ahrens left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@Ruchika-2003 I'm approving this now to keep everything in sync, but I do think we'll need to make some performance improvements later.

@willow-ahrens
willow-ahrens merged commit 30d8448 into main Jul 24, 2026
16 checks passed
yusheng036 pushed a commit that referenced this pull request Jul 24, 2026
* pulling main

* implementing and testing sampling estimator

* Changes made to sampling_stats and tests added

* Removing poetry.lock

* matching masks for same dimension

* implemented estimators from the stats paper

* updates sampling_stats without tensordef

* use_coverage implemented in sampling_stats

* heuristic flag removed

* added good1 and silly sampling

* resolving mypy errors

* reolving errors

---------

Co-authored-by: Willow Ahrens <willow@csail.mit.edu>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants